package tools

import (
	"bufio"
	"fmt"
	"os"
	"time"
)

// 文件 文件夹管理

// GetAllFile 读取当前文件夹下所有的文件,不包含文件夹的
func GetAllFile(pathname string) ([]string, error) {
	// 例如 list, err := tools.GetAllFile("runtime/")
	var s []string
	files, err := os.ReadDir(pathname)
	if err != nil {
		fmt.Println(err)
		return s, err
	}
	for _, f := range files {
		//fmt.Println(f.Name())
		s = append(s, f.Name())
	}
	return s, nil
}

// OpenFile 打开文件
func OpenFile(path string) (*os.File, error) {
	file, err := os.Open(path)
	if err != nil {
		return nil, err
	}
	return file, nil
}

// OpenFileOrCreateFile 打开文件,不存在就创建文件
func OpenFileOrCreateFile(path string) (*os.File, error) {
	file, err := OpenFile(path)
	if err != nil && os.IsNotExist(err) {
		fmt.Println(err)
		file, err = os.Create(path)
		if err != nil {
			return nil, err
		}
	}
	return file, nil
}

// LogStringToFile 写入文件日志数据
func LogStringToFile(str string, fileName string) {
	if fileName == "" {
		fileName = fmt.Sprintf("runtime/log/log_%s.txt", time.Now().Format("20060102"))
	} else {
		fileName = fmt.Sprintf("runtime/log/%s_%s.txt", fileName, time.Now().Format("20060102"))
	}
	f, err := OpenFile(fileName)
	if err != nil && os.IsNotExist(err) {
		fmt.Println(err)
		f, err = os.Create(fileName)
		_, err2 := f.WriteString(str)
		if err2 != nil {
			fmt.Println("初次写入日志失败", err2)
			return
		}
	} else {
		f, err = os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
		if err != nil {
			fmt.Println("打开文件失败", err)
		}
		write := bufio.NewWriter(f)
		logStr := fmt.Sprintf("%s--%s", time.Now().Format(FormatDateTime), str)
		_, err := write.WriteString(logStr + "\n")
		if err != nil {
			fmt.Println("追加失败", err)
			return
		}
		err = write.Flush()
		if err != nil {
			fmt.Println("追加写入失败", err)
			return
		}
	}
}
